package com.demo.util;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.util.Properties;
import com.jcraft.jsch.Channel;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;

public class SFTPUtil {

	/**
	 * 连接sftp服务器
	 * 
	 * @param host     主机
	 * @param port     端口
	 * @param username 用户名
	 * @param password 密码
	 * @return
	 */
	public ChannelSftp connect(String host, int port, String username, String password) {
		ChannelSftp sftp = null;
		try {
			JSch jsch = new JSch();
			jsch.getSession(username, host, port);
			Session sshSession = jsch.getSession(username, host, port);
			System.out.println("Session created.");
			sshSession.setPassword(password);
			Properties sshConfig = new Properties();
			sshConfig.put("StrictHostKeyChecking", "no");
			sshSession.setConfig(sshConfig);
			sshSession.connect();
			System.out.println("Session connected.");
			Channel channel = sshSession.openChannel("sftp");
			System.out.println("Opening Channel.");
			channel.connect();
			sftp = (ChannelSftp) channel;
			System.out.println("Connected to " + host + ".");

		} catch (Exception e) {
			System.err.println("erro:连接失败请检查ip及用户名密码");

		}
		return sftp;
	}

	/**
	 * 上传文件
	 * 
	 * @param directory  上传的目录
	 * @param uploadFile 要上传的文件
	 * @param sftp
	 */
	public void upload(String directory, String uploadFile, ChannelSftp sftp) {
		try {
			sftp.cd(directory);
			File file = new File(uploadFile);
			sftp.put(new FileInputStream(file), file.getName());
			System.out.println("Upload Success.");
		} catch (Exception e) {
			System.err.println("erro:请检查上传路径及文件名称");
			// e.printStackTrace();
		}
	}

	/**
	 * 下载文件
	 * 
	 * @param directory    下载目录
	 * @param downloadFile 下载的文件
	 * @param saveFile     存在本地的路径
	 * @param sftp
	 */
	public void download(String directory, String downloadFilename, String saveFile, ChannelSftp sftp) {
		try {
			sftp.cd(directory);
			File filepath = new File(saveFile);
			// 判断路径是否存在，如果不存在就创建一个
			if (!filepath.getParentFile().exists()) {
				filepath.getParentFile().mkdirs();
			}
			sftp.get(downloadFilename, new FileOutputStream(filepath));
			System.out.println("download Success.");
		} catch (Exception e) {
			System.err.println("erro:请检查下载路径及文件名称");
			// e.printStackTrace();
		}
	}

	/**
	 * 删除文件
	 * 
	 * @param directory  要删除文件所在目录
	 * @param deleteFile 要删除的文件
	 * @param sftp
	 */
	public void delete(String directory, String deleteFile, ChannelSftp sftp) {
		try {
			sftp.cd(directory);
			sftp.rm(deleteFile);
			System.out.println("delete Success.");
		} catch (Exception e) {
			System.err.println("erro:请检查删除路径及文件名称");
			// e.printStackTrace();
		}
	}

	public void disconnect(ChannelSftp sftp) {
		if (null != sftp) {
			sftp.disconnect();

			try {
				if (null != sftp.getSession()) {
					sftp.getSession().disconnect();
					System.out.println("Connection disconnection.");
				}
			} catch (JSchException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
		}
	}
}
